python assignment if else